September 2023 Imputation scores
Jeffrey Näf, Meta-Lina Spohn, Loris Michel, Nicolai Meinshausen
Author Affiliations +
Ann. Appl. Stat. 17(3): 2452-2472 (September 2023). DOI: 10.1214/22-AOAS1727

Abstract

Given the prevalence of missing data in modern statistical research, a broad range of methods is available for any given imputation task. How does one choose the “best” imputation method in a given application? The standard approach is to select some observations, set their status to missing, and compare prediction accuracy of the methods under consideration of these observations. Besides having to somewhat artificially mask observations, a shortcoming of this approach is that imputations based on the conditional mean will rank highest if predictive accuracy is measured with quadratic loss. In contrast, we want to rank highest an imputation that can sample from the true conditional distributions. In this paper we develop a framework called “Imputation Scores” (I-Scores) for assessing missing value imputations. We provide a specific I-Score, based on density ratios and projections, that is applicable to discrete and continuous data. It does not require to mask additional observations for evaluations and is also applicable if there are no complete observations. The population version is shown to be proper in the sense that the highest rank is assigned to an imputation method that samples from the correct conditional distribution. The propriety is shown under the missing completely at random (MCAR) assumption but is also shown to be valid under missing at random (MAR) with slightly more restrictive assumptions. We show empirically on a range of data sets and imputation methods that our score consistently ranks true data high(est) and is able to avoid pitfalls usually associated with performance measures such as RMSE. Finally, we provide the R-package Iscores available on CRAN with an implementation of our method.

Acknowledgements

The authors would like to thank the anonymous referees, an Associate Editor, and the Editor for their constructive comments that improved the quality of the paper.

Jeffrey Näf and Meta-Lina Spohn contributed equally to the writing of this paper.

Citation

Download Citation

Jeffrey Näf. Meta-Lina Spohn. Loris Michel. Nicolai Meinshausen. "Imputation scores." Ann. Appl. Stat. 17 (3) 2452 - 2472, September 2023. https://doi.org/10.1214/22-AOAS1727

Information

Received: 1 June 2021; Revised: 1 October 2022; Published: September 2023
First available in Project Euclid: 7 September 2023

MathSciNet: MR4637675
Digital Object Identifier: 10.1214/22-AOAS1727

Keywords: KL-divergence , Random forest , random projections , ranking , tree ensembles

Rights: Copyright © 2023 Institute of Mathematical Statistics

Vol.17 • No. 3 • September 2023
Back to Top